package com.will.nio.selector;

import lombok.extern.slf4j.Slf4j;

import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.channels.SelectionKey;
import java.nio.channels.Selector;
import java.nio.channels.ServerSocketChannel;
import java.util.Iterator;

/**
 * @author Will
 * @description:
 *              Selector    -   不处理 accept，调用 cancel 方法
 *              cancel 会取消注册在 selector 上的 channel，并从 selector 的 keys 中删除对应的 SelectionKey，后续不再监听事件
 * @date 2023-04-13 19:34
 * @version: 1.0
 */
@Slf4j
public class CancelSelectorServer {
    public static void main(String[] args) throws IOException {
        // 1. 创建 selector，管理多个 channel
        Selector selector = Selector.open();
        ServerSocketChannel ssc = ServerSocketChannel.open();
        ssc.configureBlocking(false);
        // 2. 建立 selector 和 channel 的联系（注册），channel 必须工作在非阻塞模式下
        // SelectionKey 就是将来事件发生后，通过它可以知道什么事件和哪个 channel 发生的事件
        SelectionKey sscKey = ssc.register(selector, 0, null);
        log.debug("register key:{}", sscKey);
        // 只关注 accept 事件
        sscKey.interestOps(SelectionKey.OP_ACCEPT);
        ssc.bind(new InetSocketAddress(8080));
        while (true) {
            // 3. select 方法，没有事件发生，线程阻塞，有事件，线程才会恢复运行
            // select 方法，有事件发生，但未处理，线程不会阻塞，若该事件取消，则线程阻塞
            // 因此事件发生后，要么取消，要么处理
            selector.selectNow();

            log.debug("size1:{}", selector.keys().size());
            log.debug("size2:{}", selector.selectedKeys().size());

            // 4. 处理事件，selectedKeys 内部包含了所有发生的事件
            Iterator<SelectionKey> iterator = selector.selectedKeys().iterator();
            while (iterator.hasNext()) {
                SelectionKey key = iterator.next();
                log.debug("key:{}", key);
//                ServerSocketChannel channel = (ServerSocketChannel)key.channel();
//                SocketChannel sc = channel.accept();
//                log.debug("{}", sc);
                key.cancel();
            }
        }
    }
}
